HLA: Expressive Hybrid Linear Attention via Chunk-Wise Dynamic Mixing
HLA:基于分块动态混合的高表达混合线性注意力
🧭 高效长上下文注意力机制 ★ 深度解读
Monash University · Zhejiang University
将历史 chunk 表示为仿射转移并用查询相关门控动态混合,提升长上下文检索
📌 工作参考价值(4 点)
- 对长序列用户行为/会话建模:可将历史按固定长度切 chunk,用 self-attentive pooling 得到 chunk 代表向量,再基于当前 query/item state 计算 sigmoid 门控,稀疏选择相关历史 chunk;论文中 chunk size 256、pooling window 32 是最优 trade-off,可作为初始超参。
- 工程上:HLA 用 chunk 级仿射转移 (A_j,B_j) 压缩历史,缓存规模为 N(d^2+Pd),chunk=256 时比匹配 KV cache 省 50%;推理时权重<0.1 直接置零且不重新归一化,可跳过零权重状态转移,适合长会话推荐或 Agent 记忆系统做稀疏检索。
- 训练目标:用 effective support(Hill number)构造 L_budget/L_support,鼓励每个 query 只路由到少量历史 chunk,提升推理稀疏性和长程泛化;业务上可只训练 router、冻结 backbone,低成本改造已有 GDN/线性注意力模型。
- 局限:论文未评测 Agent 多轮工具调用、持续记忆等场景,且推理延迟增加约 10–19%;若用于长程 Agent 记忆或电商超长上下文排序,需自行补充多轮、跨任务、工具反馈评测。